Python推导式

共计 1148 个字符，预计需要花费 3 分钟才能阅读完成。

推导式是 Python 中提供的一个非常方便的功能，可以通过一行代码实现创建 list、dict、tuple、set 的同时初始化一些值。

num_list = [i for i in range(5)] # 列表
print(num_list) # 结果为：[0, 1, 2, 3, 4]

num_set = {i for i in range(5)} # 集合
print(num_set) # 结果为：{0, 1, 2, 3, 4}

num_dict = {i:i for i in range(5)} # 字典
print(num_dict) # 结果为：{0: 0, 1: 1, 2: 2, 3: 3, 4: 4}

data = (i for i in range(5)) # 元组
print(data) # 结果为：<generator object <genexpr> at 0x00000>
for item in data:
    print(item)

元组不同于其他类型，不会立即执行内部循环去生成数据，而是得到一个生成器。

嵌套

## 生成一副扑克牌
poker_list = [[color, num] for num in range(1, 14) for color in [" 红桃 ", " 黑桃 ", " 方片 ", " 梅花 "]]
print(poker_list)

烧脑面试题

def num():
    return [lambda x: i * x for i in range(4)]

## 1. 执行 num() 并获取返回值 [ 函数, 函数, 函数, 函数] i=3
## 2. for 循环返回值
## 3. 返回值的每个元素 (2)
result = [m(2) for m in num()]  # [6,6,6,6]
def num():
    return (lambda x: i * x for i in range(4))

## 1. 执行 num() 并获取返回值，得到一个生成器对象
## 2. for 循环返回值
## 3. 返回值的每个元素 (2)
result = [m(2) for m in num()] # [0,2,4,6]

首字母配对

girls = ['Python', 'Java', 'C']
boys = ['Java', 'Python', 'C']
letterGirls = {}
for girl in girls:
    letterGirls.setdefault(girl[0], []).append(girl)
print([b + '+' + g for b in boys for g in letterGirls[b[0]]])

输出结果如下：

['Java + Java', 'Python + Python', 'C + C']

letterGirls.setdefault(girl[0], []) 是在 letterGirls 字典中寻找 girl[0] 元素（即 girl 变量的首字母），如果没找到，就返回默认值空列表。

那么，letterGirls.setdefault(girl[0], []) 这部分可看作 []，即可理解为 [].append(girl)。